from caffe2.python import core, workspace

print(workspace.NumCudaDevices())
